package com.xuecheng.content.service.impl;

import cn.hutool.core.bean.BeanUtil;
import com.xuecheng.content.mapper.CourseCategoryMapper;
import com.xuecheng.content.model.dto.CourseCategoryTreeDTO;
import com.xuecheng.content.model.po.CourseCategory;
import com.xuecheng.content.service.ICourseCategoryService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;

import java.awt.geom.RectangularShape;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;

/**
 * <p>
 * 课程分类 服务实现类
 * </p>
 *
 * @author 乂氼
 * @since 2024-12-18
 */
@Service
@RequiredArgsConstructor
public class CourseCategoryServiceImpl extends ServiceImpl<CourseCategoryMapper, CourseCategory> implements ICourseCategoryService {

    private final CourseCategoryMapper courseCategoryMapper; // 课程分类mapper

    /**
     * 根据id查询课程分类的子分类，不包括该分类
     *
     * @param id id
     * @return 课程分类
     */
    @Override
    public List<CourseCategoryTreeDTO> queryTreeNodes(String id) {
        // 1. 查询所有分类
        List<CourseCategoryTreeDTO> categoryList = courseCategoryMapper.queryTreeNodes(id);
        // 2. 准备返回结果集合
        List<CourseCategoryTreeDTO> result = new ArrayList<>();
        // 3. 将分类集合转换为map，key为id，value为分类对象
        Map<String, CourseCategoryTreeDTO> map = categoryList
                .stream()
                .collect(Collectors.toMap(CourseCategory::getId, category -> category));
        // 4. 遍历分类集合，将每个分类对象添加到对应的父分类对象中
        categoryList.forEach(category -> { // 要用List集合遍历D！Map存放后遍历的其实是内部的哈希表，显然顺序被打乱了。
            // 4.1 如果当前分类对象的父id等于id，则添加到返回结果集合中
            if (Objects.equals(category.getParentid(), id)) {
                result.add(map.get(category.getId()));
            }
            // 4.2 如果当前分类对象的父id不等于id，则添加到父分类对象的children集合中
            else {
                CourseCategoryTreeDTO parent = map.get(category.getParentid());
                if (parent.getChildrenTreeNodes() == null) parent.setChildrenTreeNodes(new ArrayList<>());
                parent.getChildrenTreeNodes().add(map.get(category.getId()));
            }
        });
        // 5. 返回结果
        return result;
    }
}
